TAGs: Data Parallel PyTorch